package cn.com.util;

import org.apache.flink.util.ExceptionUtils;
import org.apache.flink.util.StringUtils;

import java.io.BufferedReader;
import java.io.FileReader;
import java.util.ArrayList;
import java.util.List;

/**
 * SQL文件内容解析
 *
 * @author WZQ
 * @date 2021-08-27 21:32
 **/
public class SqlFileUtil {


    /**
     * 获取指定sql文件内容，并按照英文分号分隔，单独存储每个sql语句，同时去掉使用--注释的内容。
     *
     * @param filepath sql文件绝对路径
     * @return 每个sql语句组成的列表
     */
    public static List<String> getFileContentAsSql(String filepath) {
        List<String> lines = readFile(filepath);
        List<String> sql = new ArrayList<>(256);
        StringBuilder oneSql = new StringBuilder(256);
        if (lines.size() > 0) {
            for (String line : lines) {
                line = line.trim().replaceAll("--.*", "");
                if (StringUtils.isNullOrWhitespaceOnly(line)) {
                    continue;
                }
                if (line.endsWith(";")) {
                    oneSql.append(line.replace(";", "").trim());
                    if (oneSql.length() > 0) {
                        sql.add(oneSql.toString().trim());
                        oneSql.delete(0, oneSql.length());
                    }
                } else {
                    oneSql.append(line).append("\n");
                }
            }
        }
        System.out.println("\n处理之后的sql内容为：");
        for (String s : sql) {
            System.out.println(s + "\n;\n");
        }
        System.out.println();
        return sql;
    }

    /**
     * 按行读取给定文件内容
     *
     * @param filePath 文件路径
     * @return 文件所有行内容组成的列表
     */
    private static List<String> readFile(String filePath) {
        List<String> result = new ArrayList<>();
        try {
            BufferedReader reader = new BufferedReader(new FileReader(filePath));
            String line;
            while ((line = reader.readLine()) != null) {
                result.add(line);
            }
        } catch (Exception e) {
            throw new RuntimeException("未找到指定的sql文件：\n" + ExceptionUtils.stringifyException(e));
        }
        System.out.println("\n读取到的sql文件内容为：");
        for (String s : result) {
            System.out.println(s);
        }
        System.out.println();
        return result;
    }

}
